'''
@Project ：hdcms-scripts
@File    ：room_grouping.py
@IDE     ：PyCharm
@Author  ：朱广杰
@Date    ：2024/11/6 16:22
'''
import os
import time
from pathlib import Path

from common.util import cmd_util
from common.hd_log import mylog


def room_grouping(course_id: str):
    # current_file_path = os.path.abspath(__file__)
    # path_obj = Path(current_file_path)
    # project_path = path_obj.parents[2]
    # script_path = os.path.join(project_path, 'data')
    # mylog.info(f'jmeter 脚本路径：{script_path}')
    # cmd_str = f'cd {script_path} && rm -rf room_grouping.jtl && jmeter -n -t ./room_grouping.jmx -Jcourse_id={course_id} -l room_grouping.jtl'
    # cmd_util.cmd(cmd_str)
    # is_exists = cmd_util.check_file_exists(script_path, 'room_grouping.jtl', 60)
    # if is_exists is False:
    #     mylog.error(f'jmeter 脚本路径：{script_path}下没有 room_grouping.jtl 文件，请检查脚本是否执行成功')
    #     raise RuntimeError('jmeter 执行脚本失败')
    # mylog.info('脚本执行成功')
    # return True
    current_file_path = os.path.abspath(__file__)
    path_obj = Path(current_file_path)
    project_path = path_obj.parents[2]
    script_path = os.path.join(project_path, 'data')
    mylog.info(f'jmeter 脚本路径：{script_path}')
    cmd_str1 = f'cd {script_path} && rm -rf room_grouping.jtl && rm -rf jmeter.log'
    cmd_util.cmd(cmd_str1)
    try:
        is_deleted = cmd_util.check_file_exists(script_path, 'room_grouping.jtl', 3)
        if is_deleted is True:
            mylog.info('room_grouping.jtl 文件已删除')
    except Exception as e:
        mylog.error(f'{e}')
    cmd_str2 = f'cd {script_path} && jmeter -n -t ./room_grouping.jmx -Jcourse_id={course_id} -l room_grouping.jtl'
    cmd_util.cmd(cmd_str2)
    is_exists = cmd_util.check_file_exists(script_path, 'room_grouping.jtl', 60)
    if is_exists is False:
        mylog.error(f'jmeter 脚本路径：{script_path}下没有 room_grouping.jtl 文件，请检查脚本是否执行成功')
        raise RuntimeError(f'jmeter 脚本路径：{script_path}下没有 room_grouping.jtl 文件，请检查脚本是否执行成功')
    mylog.info('脚本执行成功')
    return True
